package com.airena.aiservice;


import dev.langchain4j.service.MemoryId;
import dev.langchain4j.service.SystemMessage;
import dev.langchain4j.service.UserMessage;
import dev.langchain4j.service.V;
import dev.langchain4j.service.spring.AiService;
import dev.langchain4j.service.spring.AiServiceWiringMode;
import reactor.core.publisher.Flux;

@AiService(
        wiringMode = AiServiceWiringMode.EXPLICIT,//手动装配
        chatModel = "openAiChatModel", //指定模型
        streamingChatModel = "openAiStreamingChatModel", //指定流式模型
        chatMemory = "chatMemory",//配置会话记忆对象
        chatMemoryProvider = "chatMemoryProvider"//配置会话记忆提供者
//        contentRetriever = "contentRetriever"//配置向量数据库检索对象
)
public interface ConsultantService {

//    系统提示词
//    @SystemMessage(fromResource = "system.txt")
    @SystemMessage("你是一个利用 AI 来做角色扮演的网站，用户可以搜索自己感兴趣的角色并可与其进行语音聊天,你只能回答与角色符合的问题,其他问题不予回答。{{role}}")
    public Flux<String> chat(@MemoryId String memoryId, @UserMessage String message, @V("role") String roleMessage);
}